package database

import (
	"context"
	"fmt"
	"time"

	"github.com/redis/go-redis/v9"
	"github.com/sirupsen/logrus"
	"ygang.top/gin-template/internal/config"
)

func NewRedisClient(config *config.ApplicationConfig) *redis.Client {
	// 配置连接信息
	option := redis.Options{
		Addr:     fmt.Sprintf("%s:%s", config.Redis.Host, config.Redis.Port),
		Password: config.Redis.Password, // 密码
		DB:       config.Redis.DB,       // redis一共16个库，指定其中一个库即可

		//连接池容量及闲置连接数量
		PoolSize:     config.Redis.PoolSize,     // 连接池最大socket连接数，默认为4倍CPU数， 4 * runtime.NumCPU
		MinIdleConns: config.Redis.MinIdleConns, // 在启动阶段创建指定数量的Idle连接，并长期维持idle状态的连接数不少于指定数量；。

		//超时
		DialTimeout:  time.Duration(config.Redis.DialTimeout) * time.Second,  // 连接建立超时时间，默认5秒。
		ReadTimeout:  time.Duration(config.Redis.ReadTimeout) * time.Second,  // 读超时，默认3秒， -1表示取消读超时
		WriteTimeout: time.Duration(config.Redis.WriteTimeout) * time.Second, // 写超时，默认等于读超时
		PoolTimeout:  time.Duration(config.Redis.PoolTimeout) * time.Second,  // 当所有连接都处在繁忙状态时，客户端等待可用连接的最大等待时长，默认为读超时+1秒。

		//命令执行失败时的重试策略
		MaxRetries:      config.Redis.MaxRetries,                                        // 命令执行失败时，最多重试多少次，默认为0即不重试
		MinRetryBackoff: time.Duration(config.Redis.MinRetryBackoff) * time.Millisecond, // 每次计算重试间隔时间的下限，默认8毫秒，-1表示取消间隔
		MaxRetryBackoff: time.Duration(config.Redis.MaxRetryBackoff) * time.Millisecond, // 每次计算重试间隔时间的上限，默认512毫秒，-1表示取消间隔
	}
	// 获取客户端实例
	client := redis.NewClient(&option)
	// 使用 Ping() 方法测试是否成功连接到 Redis 服务器
	ctx := context.Background()
	pong, err := client.Ping(ctx).Result()
	if err != nil {
		logrus.Fatalf("Redis连接失败: %s\n", err.Error())
		return nil
	}
	logrus.Infof("连接到 Redis Ping: %s\n", pong)
	return client
}
